package com.lukebang.util;

import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;

import javax.imageio.ImageIO;

import org.springframework.web.multipart.MultipartFile;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;

@SuppressWarnings("restriction")
public class Base64ImageUtils {
	/**
	 * 将网络图片进行Base64位编码
	 * 
	 * @param imageUrl
	 *            图片的url路径，如http://.....xx.jpg
	 * @return
	 */
	public static String encodeImgageToBase64(URL imageUrl) {// 将图片文件转化为字节数组字符串，并对其进行Base64编码处理
		ByteArrayOutputStream outputStream = null;
		try {
			BufferedImage bufferedImage = ImageIO.read(imageUrl);
			outputStream = new ByteArrayOutputStream();
			ImageIO.write(bufferedImage, "jpg", outputStream);
		} catch (MalformedURLException e1) {
			e1.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		// 对字节数组Base64编码
		BASE64Encoder encoder = new BASE64Encoder();
		return encoder.encode(outputStream.toByteArray());// 返回Base64编码过的字节数组字符串
	}

	/**
	 * 将本地图片进行Base64位编码
	 * 
	 * @param imageFile
	 *            图片的url路径，如F:/.....xx.jpg
	 * @return
	 * @throws Exception
	 */
	public static String encodeImgageToBase64(MultipartFile imageFile) throws Exception {// 将图片文件转化为字节数组字符串，并对其进行Base64编码处理
		BASE64Encoder encoder = new BASE64Encoder();
		return encoder.encode(imageFile.getBytes());// 返回Base64编码过的字节数组字符串
	}

	/**
	 * 将Base64位编码的图片进行解码，并保存到指定目录
	 * 
	 * @param base64
	 *            base64编码的图片信息
	 * @return
	 */
	public static void decodeBase64ToImage(String base64, String path, String imgName) {
		BASE64Decoder decoder = new BASE64Decoder();
		try {
			FileOutputStream write = new FileOutputStream(new File(path + imgName));
			byte[] decoderBytes = decoder.decodeBuffer(base64);
			write.write(decoderBytes);
			write.close();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	/**
	 * 将Base64位编码的图片进行解码，并保存到指定目录
	 * 
	 * @param base64
	 *            base64编码的图片信息
	 * @return
	 * @return
	 */
	public static byte[] decodeBase64ToImage(String base64) {
		BASE64Decoder decoder = new BASE64Decoder();
		try {
			byte[] decoderBytes = decoder.decodeBuffer(base64);
			return decoderBytes;
		} catch (IOException e) {
			e.printStackTrace();
			return null;
		}
	}

	public static void main(String[] args) {
		URL url = null;
		try {
			url = new URL("http://www.faceplusplus.com.cn/wp-content/themes/faceplusplus/assets/img/demo/9.jpg");
		} catch (MalformedURLException e) {
			e.printStackTrace();
		}
		String encoderStr = Base64ImageUtils.encodeImgageToBase64(url);
		System.out.println(encoderStr);

		Base64ImageUtils.decodeBase64ToImage(encoderStr, "E:/", "football.jpg");

	}
}